package com.zjsru.plan2024.oneday;

import java.util.*;

/**
 * 2831. 找出最长等值子数组
 * @Author: cookLee
 * @Date: 2024-05-23
 */
public class LongestEqualSubarray {

    /**
     * 主
     * \
     * 如果子数组中所有元素都相等，则认为子数组是一个 等值子数组 。注意，空数组是 等值子数组 。
     * \
     * 输入：nums = [1,3,2,3,1,3], k = 3
     * 输出：3
     * 解释：最优的方案是删除下标 2 和下标 4 的元素。
     * 删除后，nums 等于 [1, 3, 3, 3] 。
     * 最长等值子数组从 i = 1 开始到 j = 3 结束，长度等于 3 。
     * 可以证明无法创建更长的等值子数组。
     * \
     * 输入：nums = [1,1,2,2,1,1], k = 2
     * 输出：4
     * 解释：最优的方案是删除下标 2 和下标 3 的元素。
     * 删除后，nums 等于 [1, 1, 1, 1] 。
     * 数组自身就是等值子数组，长度等于 4 。
     * 可以证明无法创建更长的等值子数组。
     * \
     * @param args args
     */
    public static void main(String[] args) {
        List<Integer> nums = Arrays.asList(1,3,2,3,1,3);
        int k = 3;
        LongestEqualSubarray longestEqualSubarray = new LongestEqualSubarray();
        System.out.println(longestEqualSubarray.longestEqualSubarray(nums, k));
    }


    /**
     * 最长相等子阵列
     *
     * @param nums nums
     * @param k    k
     * @return int
     */
    public int longestEqualSubarray(List<Integer> nums, int k) {
        int n = nums.size();
        Map<Integer, List<Integer>> map = new HashMap<>();
        for (int i = 0; i < n; i++) {
            map.computeIfAbsent(nums.get(i), x -> new ArrayList<>()).add(i);
        }
        int ans = 0;
        for (List<Integer> vec : map.values()) {
            for (int i = 0, j = 0; i < vec.size(); i++) {
                //缩小窗口，直到不同元素数量小于等于 k
                while (vec.get(i) - vec.get(j) - (i - j) > k) {
                    j++;
                }
                ans = Math.max(ans, i - j + 1);
            }
        }
        return ans;
    }
}
